package cn.edu.nwafu.nic.nucleic_acid_test.config;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.support.http.StatViewServlet;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;
import java.util.HashMap;

/*
 * @author: SungLee
 * @date: 2022-02-19 14:21
 * @description: Druid数据源配置
 */

@Configuration
public class DruidConfig {

    // 绑定yml中的配置
    @ConfigurationProperties(prefix = "spring.datasource.druid")
    @Bean
    public DataSource druidDataSourece() {
        return new DruidDataSource();
    }

    // 因为SpringBoot内置了servlet容器，所以没有web.xml
    // 替代方法：ServletRegistrationBean注册进去
    @Bean
    public ServletRegistrationBean statViewServlet() {
        ServletRegistrationBean<StatViewServlet> bean = new
                ServletRegistrationBean<>(new StatViewServlet(), "/druid/*");

        // 后台需要有人登录,设置账号密码
        HashMap<String, String> initParameters = new HashMap<>();

        // 增加配置
        // 登录key 固定的    loginUsername       loginPassword
        initParameters.put("loginUsername", "admin");
        initParameters.put("loginPassword", "123456");


        /* 允许谁可以访问
         * "" 表示所有人
         * "localhost"表示本机可以访问,
         * 还可以指定某个人
         * 想要禁止访问，""指定ip即可  例如：initParameters.put("xiaozhang","192.168.18.176"); 即可
         */
        initParameters.put("allow", "");

        // 设置初始化参数
        bean.setInitParameters(initParameters);
        return bean;
    }

}
